data_path=$1

if [ ! $INPUT_PATH ]; then
    python3.7 -u main.py    --mode train --folder_dir stargan_full_1p \
                            --gpus 1 --batch_size 16 --epoch 50 \
                            --selected_attrs Black_Hair Blond_Hair Brown_Hair Male Young
else
    python3.7 -u main.py    --mode train --folder_dir stargan_full_1p \
                            --gpus 1 --batch_size 16 --epoch 50 \
                            --selected_attrs Black_Hair Blond_Hair Brown_Hair Male Young \
                            --celeba_image_dir $INPUT_PATH 
fi 